pytorchのインストールとアンインストール
1つ、pytorchのインストール
gpuバージョンのtensorflowをインストールしました
1.1最初にcmdを開いて、自分でインストールしたcudaのバージョンを確認します
入る
nvcc --version
cuda10.0であることがわかります
1.2オフラインダウンロードパッケージを見つけてインストールします
pytorchオフラインダウンロードアドレス:pytorchオフラインダウンロード
Windowsシステム+ Pythonバージョン-cudaバージョンに従ってダウンロード
1.3pipを使用してインストールする
pip install torch-1.0.0-cp36-cp36m-win_amd64.whl
トーチ1.0.0に対応するトーチビジョンのバージョンは0.2.0です。
pip install torchvision==0.2.0
1.4GPUバージョンが正常にインストールされていることを確認します
テスト手順は次のとおりです。
import torch
print(torch.__version__)
print(torch.cuda.is_available())
演算結果:
1.0.0
True
PyTorchのバージョンは1.0.0Trueです
。これは、GPUバージョンが正常にインストールされたことを意味します。
2、pytorchのアンインストール
1.condaを使用してPytorchをアンインストールします
conda uninstall pytorch
conda uninstall libtorch
2.pipを使用してPytorchをアンインストールします
pip uninstall torch
pytorchがcondainstallコマンドでインストールされた場合は、最初の方法を使用します。pipinstallコマンドでインストールされた場合は、2番目の方法を使用します。または、両方の方法を試してください。