iru on Nostr: `nix develop` > Conda, venv, toolbox/distrobox ``` { description = “Instant ROCM ...
`nix develop` > Conda, venv, toolbox/distrobox
```
{
description = “Instant ROCM 5.7 environment”;
inputs.nixpkgs.url = "github:nixos/nixpkgs/nixos-23.11";
outputs = { self, nixpkgs }:
let
pkgs = import nixpkgs {
system = "x86_64-linux";
};
pythonPackages = pkgs.python310Packages;
pyPkgs = with nixpkgs; [
(pkgs.python310.withPackages(ps: with ps; [
pip
torchWithRocm
]))
];
in
{
devShells.x86_64-linux = {
default = pkgs.mkShell {
buildInputs = [
pyPkgs
pkgs.radeontop
pkgs.nvtop-amd
pkgs.rocmPackages.clr
pkgs.rocmPackages.hipcc
pkgs.rocmPackages.miopen
pkgs.rocmPackages.rocblas
pkgs.rocmPackages.hipblas
pkgs.rocmPackages.rocminfo
pkgs.rocmPackages.rocm-smi
pkgs.rocmPackages.llvm.llvm
pkgs.rocmPackages.llvm.clang
pkgs.rocmPackages.llvm.bintools
pkgs.rocmPackages.llvm.clang-tools-extra
];
};
};
};
}
```
```
{
description = “Instant ROCM 5.7 environment”;
inputs.nixpkgs.url = "github:nixos/nixpkgs/nixos-23.11";
outputs = { self, nixpkgs }:
let
pkgs = import nixpkgs {
system = "x86_64-linux";
};
pythonPackages = pkgs.python310Packages;
pyPkgs = with nixpkgs; [
(pkgs.python310.withPackages(ps: with ps; [
pip
torchWithRocm
]))
];
in
{
devShells.x86_64-linux = {
default = pkgs.mkShell {
buildInputs = [
pyPkgs
pkgs.radeontop
pkgs.nvtop-amd
pkgs.rocmPackages.clr
pkgs.rocmPackages.hipcc
pkgs.rocmPackages.miopen
pkgs.rocmPackages.rocblas
pkgs.rocmPackages.hipblas
pkgs.rocmPackages.rocminfo
pkgs.rocmPackages.rocm-smi
pkgs.rocmPackages.llvm.llvm
pkgs.rocmPackages.llvm.clang
pkgs.rocmPackages.llvm.bintools
pkgs.rocmPackages.llvm.clang-tools-extra
];
};
};
};
}
```